PoseDetection

sealed class PoseDetection<U : InferenceModel> : OnnxModelType<U>

Pose detection models.

Types

MoveNetSinglePoseLighting

android

object MoveNetSinglePoseLighting : ONNXModels.PoseDetection<SinglePoseDetectionModel>

This model is a convolutional neural network model that runs on RGB images and predicts human joint locations of a single person. (edges are available in org.jetbrains.kotlinx.dl.onnx.inference.posedetection.edgeKeyPointsPairs and keypoints are in org.jetbrains.kotlinx.dl.onnx.inference.posedetection.keyPoints).

MoveNetSinglePoseThunder

android

object MoveNetSinglePoseThunder : ONNXModels.PoseDetection<SinglePoseDetectionModel>

This model is a convolutional neural network model that runs on RGB images and predicts human joint locations of a single person. (edges are available in org.jetbrains.kotlinx.dl.onnx.inference.posedetection.edgeKeyPointsPairs and keypoints are in org.jetbrains.kotlinx.dl.onnx.inference.posedetection.keyPoints).

Functions

model

android

open fun model(modelHub: ModelHub): OnnxInferenceModel

pretrainedModel

android

abstract fun pretrainedModel(modelHub: ModelHub): U

Properties

inputShape

android

open val inputShape: LongArray?

Shape of the input accepted by this model, without batch size.

modelRelativePath

android

open override val modelRelativePath: String

preprocessor

android

open val preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>

Inheritors

ONNXModels.PoseDetection

ONNXModels.PoseDetection

ONNXModels.PoseDetection